package cluster

import (
	"fmt"
	"net"
	"time"
)

// NodeStatus 表示节点状态
type NodeStatus string

const (
	// NodeStatusUnknown 表示未知状态
	NodeStatusUnknown NodeStatus = "unknown"
	// NodeStatusUp 表示节点正常运行
	NodeStatusUp NodeStatus = "up"
	// NodeStatusDown 表示节点已停止
	NodeStatusDown NodeStatus = "down"
	// NodeStatusLeaving 表示节点正在离开集群
	NodeStatusLeaving NodeStatus = "leaving"
	// NodeStatusJoining 表示节点正在加入集群
	NodeStatusJoining NodeStatus = "joining"
)

// NodeInfo 表示节点信息
type NodeInfo struct {
	// ID 是节点ID
	ID string
	// Address 是节点地址
	Address string
	// Status 是节点状态
	Status NodeStatus
	// LastSeen 是最后一次看到节点的时间
	LastSeen time.Time
	// Metadata 是节点元数据
	Metadata map[string]string
	// Role 是节点角色
	Role string
	// Tags 是节点标签
	Tags []string
	// Version 是节点版本
	Version string
}

// String 返回节点信息的字符串表示
func (n *NodeInfo) String() string {
	return fmt.Sprintf("Node{ID: %s, Address: %s, Status: %s, Role: %s}", n.ID, n.Address, n.Status, n.Role)
}

// ParseAddress 解析地址字符串为主机和端口
func ParseAddress(address string) (string, int, error) {
	host, portStr, err := net.SplitHostPort(address)
	if err != nil {
		return "", 0, fmt.Errorf("invalid address format: %w", err)
	}

	// 如果主机为空，使用localhost
	if host == "" {
		host = "localhost"
	}

	// 解析端口
	var port int
	_, err = fmt.Sscanf(portStr, "%d", &port)
	if err != nil {
		return "", 0, fmt.Errorf("invalid port: %w", err)
	}

	return host, port, nil
}

// FormatAddress 格式化主机和端口为地址字符串
func FormatAddress(host string, port int) string {
	return fmt.Sprintf("%s:%d", host, port)
}

// IsLocalAddress 检查地址是否为本地地址
func IsLocalAddress(address string) bool {
	host, _, err := ParseAddress(address)
	if err != nil {
		return false
	}

	// 检查是否为localhost
	if host == "localhost" || host == "127.0.0.1" || host == "::1" {
		return true
	}

	// 获取本地IP地址
	addrs, err := net.InterfaceAddrs()
	if err != nil {
		return false
	}

	for _, addr := range addrs {
		if ipnet, ok := addr.(*net.IPNet); ok {
			if ipnet.IP.String() == host {
				return true
			}
		}
	}

	return false
}

// GetLocalIP 获取本地IP地址
func GetLocalIP() (string, error) {
	addrs, err := net.InterfaceAddrs()
	if err != nil {
		return "", err
	}

	for _, addr := range addrs {
		if ipnet, ok := addr.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
			if ipnet.IP.To4() != nil {
				return ipnet.IP.String(), nil
			}
		}
	}

	return "127.0.0.1", nil
}
